import os
import shutil

# 原始文件夹路径
source_folder = r"D:\songlin\data\8.图像分割相关\马"

# 创建目标文件夹
json_folder = os.path.join(source_folder, "json")
image_folder = os.path.join(source_folder, "image")

# 如果目标文件夹不存在，则创建
os.makedirs(json_folder, exist_ok=True)
os.makedirs(image_folder, exist_ok=True)

# 遍历文件夹中的所有文件
for filename in os.listdir(source_folder):
    file_path = os.path.join(source_folder, filename)

    # 如果是JSON文件，移动到json_files文件夹
    if filename.endswith(".json"):
        shutil.move(file_path, os.path.join(json_folder, filename))
    # 如果是图片文件（常见格式），移动到image_files文件夹
    elif filename.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')):
        shutil.move(file_path, os.path.join(image_folder, filename))

print("文件分类完成！")